package dm_schema

import (
	"gorm.io/gorm"
	"server/global"
	"server/model/common/request"
)

type DmSchemaService struct {
}

func (dmSchemaService *DmSchemaService) CreateDmSchema(dmSchema *DmSchema) (err error) {
	err = global.SysDB.Create(dmSchema).Error
	return err
}

func (dmSchemaService *DmSchemaService) DeleteDmSchema(dmSchema DmSchema) (err error) {
	err = global.SysDB.Transaction(func(tx *gorm.DB) error {
		if err := tx.Model(&DmSchema{}).Where("id = ?", dmSchema.ID).Update("deleted_by", dmSchema.DeletedBy).Error; err != nil {
			return err
		}
		if err = tx.Delete(&dmSchema).Error; err != nil {
			return err
		}
		return nil
	})
	return err
}

func (dmSchemaService *DmSchemaService) DeleteDmSchemaByIds(ids request.IdsReq, deleted_by uint) (err error) {
	err = global.SysDB.Transaction(func(tx *gorm.DB) error {
		if err := tx.Model(&DmSchema{}).Where("id in ?", ids.Ids).Update("deleted_by", deleted_by).Error; err != nil {
			return err
		}
		if err := tx.Where("id in ?", ids.Ids).Delete(&DmSchema{}).Error; err != nil {
			return err
		}
		return nil
	})
	return err
}

func (dmSchemaService *DmSchemaService) UpdateDmSchema(dmSchema DmSchema) (err error) {
	err = global.SysDB.Save(&dmSchema).Error
	return err
}

func (dmSchemaService *DmSchemaService) GetDmSchemaById(id uint) (dmSchema DmSchema, err error) {
	err = global.SysDB.Where("id = ?", id).First(&dmSchema).Error
	return
}

func (dmSchemaService *DmSchemaService) GetDmSchemaInfoList(info DmSchemaSearch) (list []DmSchema, total int64, err error) {
	limit := info.PageSize
	offset := info.PageSize * (info.Page - 1)
	db := global.SysDB.Model(&DmSchema{})
	var dmSchemas []DmSchema
	if info.StartCreatedAt != nil && info.EndCreatedAt != nil {
		db = db.Where("created_at BETWEEN ? AND ?", info.StartCreatedAt, info.EndCreatedAt)
	}
	err = db.Count(&total).Error
	if err != nil {
		return
	}

	err = db.Limit(limit).Offset(offset).Find(&dmSchemas).Error
	return dmSchemas, total, err
}

func (dmSchemaService *DmSchemaService) GetAllDmSchemas() (dmSchema []DmSchema, err error) {
	err = global.SysDB.Find(&dmSchema).Error
	return dmSchema, err
}

func (dmSchemaService *DmSchemaService) GetDmSchema() (dmSchema []DmSchema, err error) {
	err = global.SysDB.Find(&dmSchema).Error
	return dmSchema, err
}
